# app/schemas/chat.py
from pydantic import BaseModel
from typing import Optional

class ChatRequest(BaseModel):
    model: str                  # 'openai', 'wenxin', 'qwen', 'zhipu'
    message: str                # 用户输入内容
    session_id: Optional[int]   # 可选，对话会话 id

class ChatResponse(BaseModel):
    reply: str                  # AI 回复内容
